import sys
from langchain_community.llms import Ollama

llm = Ollama(model="llama3", base_url="http://localhost:11434")
def chat_with_ollama_get_tg():
    translate_count = 0
    sourceFilePath = r"D:\sources\KJVS\localgit\com-kuajingvs-app-api\src\main\resources\i18n\zh-CN.properties"
    #读取i18n文件，并取出每一行。
    with open(sourceFilePath, "r", encoding="utf-8") as f:
        lines = f.readlines()
        for line in lines:

            if "=" in line:
                # 将每一行的文本按=分割，左边为key，右边为value
                key = line.split("=")[0].strip()
                value = line.split("=")[1].strip()


                # 将value传给ollama进行翻译，将翻译后的结果与key用=号拼接，并生成新的西班牙语 i18n文件
                prompt = u"""
                   系统：你是一个翻译助手，用于将文字从中文或英文翻译为西班牙语
                   User：请翻译以下文字到西班牙语，只返回翻译后的结果，不用返回其它信息。
                   %s
                   Assistant:""" % value

                response = llm(prompt)
                print(key,value,"=>",response)

                translate_count+= 1

                if translate_count > 10:
                    break



if __name__== "__main__":
    chat_with_ollama_get_tg()